package com.webserver_v6.controller;

import com.webserver_v6.annotation.Controller;
import com.webserver_v6.annotation.RequestMapping;
import com.webserver_v6.entity.User;
import com.webserver_v6.http.HttpServletRequest;
import com.webserver_v6.http.HttpServletResponse;
import com.webserver_v6.util.DBUtil;

import java.io.*;
import java.net.URISyntaxException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

@Controller
public class UserController {
    private static File userDir;
    static {
        userDir = new File("./users");
        if(!userDir.exists()){
            userDir.mkdirs();
        }
    }



    @RequestMapping("/regUser")
    public void reg(HttpServletRequest request, HttpServletResponse response){
        System.out.println("开始处理注册!!!!");
        //获取表单信息
        String username = request.getParameters("username");
        String password = request.getParameters("password");
        String nickname = request.getParameters("nickname");
        String ageStr = request.getParameters("age");
        //必要验证
        if(username==null||username.isEmpty()||
                password==null||password.isEmpty()||
                nickname==null||nickname.isEmpty()||
                ageStr==null||ageStr.isEmpty()||!ageStr.matches("[0-9]+")
        ){
            //信息输入有误提示页面

            response.sendRedirect("reg_info_error.html");
            return;
        }

        System.out.println(username+","+password+","+nickname+","+ageStr);

        int age = Integer.parseInt(ageStr);
        try(
                Connection connection = DBUtil.getConnection();
                ) {
            String sql = "select 1 from userinfo where username=? and password=?";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,username);
            ps.setString(2,password);
            ResultSet rs = ps.executeQuery();
            if (rs.next()){
                response.sendRedirect("/have_user.html");
                return;
            }
            sql = "insert into userinfo(username,password,nickname,age) values (?,?,?,?)";
            ps = connection.prepareStatement(sql);
            ps.setString(1,username);
            ps.setString(2,password);
            ps.setString(3,nickname);
            ps.setInt(4,age);
            int number = ps.executeUpdate();
            if (number>0){
                response.sendRedirect("reg_success.html");
            }else {
                response.sendRedirect("reg_fail.html");
            }
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
    }

    @RequestMapping("/loginUser")
    public void login(HttpServletRequest request,HttpServletResponse response){
        String username = request.getParameters("username");
        String password = request.getParameters("password");

        if (username==null||username.isEmpty()||
            password==null||password.isEmpty()
          ){
            response.sendRedirect("login_info_error.html");
            return;
        }
        try (
                Connection connection = DBUtil.getConnection();
                ){
            String sql = "select id ,username, password, nickname ,age from userinfo where username=? and password=?";
            PreparedStatement ps = connection.prepareStatement(sql);
            ps.setString(1,username);
            ps.setString(2,password);
            ResultSet rs = ps.executeQuery();
            if (rs.next()){
                response.sendRedirect("login_success.html");
            }else {
                response.sendRedirect("login_fail.html");
            }
        } catch (SQLException throwables) {
            throwables.printStackTrace();
        }
    }
}

